import java.util.HashMap;
import java.util.Map;

public class a10 {
    //给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
    //
    //子数组是数组中元素的连续非空序列。
    public static int subarraySum(int[] nums, int k) {
        int sum = 0;
        int count = 0;
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        for (int num : nums) {
            sum += num;
            if(map.containsKey(sum - k)) {
                count+=map.get(sum - k);
            }
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }

        return count;
    }

    public static void main(String[] args) {
        int[] nums = {-1,-1,1};
        int k = 0;
        System.out.println(subarraySum(nums, k));
    }
}
